package com.syh.bishe.interceptor;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.syh.bishe.Util.JwtUtils;
import com.syh.bishe.Util.UserHolder;
import com.syh.bishe.pojo.UserInfo;
import com.syh.bishe.service.ICompaniesService;
import com.syh.bishe.service.UserService;
import io.jsonwebtoken.Claims;
import jakarta.annotation.Resource;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;

/**
 * <p>
 *在目标资源方法执行前执行的拦截器方法。
 * </p>
 *
 * @author syh
 * @since 2024-05-08
 */
@Component
@Slf4j
public class LoginCheckInterceptor implements HandlerInterceptor {
    @Resource
    private UserService userService;
    @Resource
    private ICompaniesService companiesService;
    /**
     * 在目标资源方法执行前执行的拦截器方法。
     *
     * @param request  HttpServletRequest对象，代表客户端的HTTP请求
     * @param response HttpServletResponse对象，用于向客户端发送响应
     * @param handler  将要执行的处理器对象
     * @return boolean 返回值。如果返回true，表示请求放行，继续执行后续逻辑；如果返回false，表示请求被拦截，不继续执行。
     * @throws Exception 抛出异常，用于处理预处理过程中的错误情况。
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        // 获取请求头中的token
        String tokens = request.getHeader("authorization");
        response.setContentType("application/json;charset = UTF-8");

        ObjectMapper mapper = new ObjectMapper();

        // 判断token是否存在或为空
        if (tokens == null || "".equals(tokens)) {
            // 如果token不存在或为空，则记录日志并返回false，阻止请求继续执行
            log.info("token不存在");
            return false;
        }
        String token = tokens.startsWith("Bearer ") ? tokens.substring(7) : tokens;
        log.info("token:" + token);
        String email = "";
        // 尝试解析token，验证其正确性
        try {
            Claims claims = JwtUtils.parseJWT(token);
            if (claims == null) {
                log.info("解析失败，claims为空");
                return false;
            }
            email = claims.get("email", String.class);
            log.info("email: " + email);

            String status = userService.getStatus(email);
            if ("0".equals(status)) {
                log.info("用户被禁用");
                return false;
            }
            UserInfo userInfo = new UserInfo();
            userInfo.setEmail(email);
            userInfo.setCompanyId(companiesService.getId(claims.get("company", String.class)));
            userInfo.setId(claims.get("id", Integer.class));
            UserHolder.saveUser(userInfo);
        } catch (Exception e) {
            // 如果解析失败，记录日志并返回false，阻止请求继续执行
            log.info("请求头不正确!!");
            return false;
        }
        log.info(email);
        //System.err.println(email);
        // 如果token存在且解析成功，则返回true，允许请求继续执行
        return true;
    }


    //目标资源方法执行后执行
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        log.info("请求完成后释放资源");
        UserHolder.removeUser(); // 请求完成后释放资源
    }
}

